Python If Else Trick